package zzzhc.image.recognize.nn;

import org.joone.engine.Layer;
import org.joone.engine.WTALayer;

public class WinnerTakeAllSOM extends AbstractSOM {

	@Override
	protected Layer createCoreLayer(NNInput nnInput) {
		WTALayer coreLayer = new WTALayer();
		coreLayer.setLayerHeight(nnInput.getCodeTypeNum());
		return coreLayer;
	}

	@Override
	protected void doSubSetNNInput() {
		((WTALayer) coreLayer).setLayerHeight(nnInput.getCodeTypeNum());
	}

}
